from db import usersys_db
from usr_pwd_check import check_pwd
from usr_pwd_check import check_usr_len
# main.py 主程序 入口
msg = """欢迎来到更高级的用户系统
请输入：0 表示 退出
请输入：1 表示 注册
请输入：2 表示 登录
请输入：3 表示 修改密码
请输入：4 表示 删除用户
"""
print(msg)
t = usersys_db()

while True:
    flag = input('请输入您的指令')
    if flag == '0':
        break
    elif flag == '1':
        username = input('请输入用户名')
        if check_usr_len(usr=username):
            # 符合长度
            if t.have_uesr(usr=username):
                print('该用户已存在')
            else:
                password = input('请输入密码')
                if check_pwd(pwd=password):
                    t.add_user(usr=username,pwd=password)
                else:
                    print('密码不符合要求')
        else:
            print('用户名不符合长度')

    elif flag == '2':
        username = input('请输入用户名:')
        if check_usr_len(usr=username):
            if t.have_uesr(usr=username):
                password = input('请输入密码：')
                db_pwd = t.get_user_pwd(usr=username)
                if password == db_pwd:
                    print('登录成功....')
                else:
                    print('用户名或密码错误')
            else:
                print('没有该用户')
        else:
            print('用户名不符合要求')
